from ultralytics import YOLOv10

model = YOLOv10(r"/home/aistudio/work/yolov10-main/work/runs/train/exp2/weights/best.pt")
model.predict(source=r"/home/aistudio/data/data315411/dataset/images/test", imgsz=640, conf=0.03, save=True)
# result = model.val(data=r"/home/aistudio/work/yolov10-main/dataset/data.yaml", imgsz=640, conf=0.03)
# print(result)
